#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <unistd.h>
#include <cstdlib>
#include <sys/wait.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/stat.h>

const int basesize=1024;
const int envnums=64;
const int argcnums=64;

//我的系统的环境变量
char* Genv[envnums];
char* Gargv[argcnums];
int Gargc;

// 全局的当前工作路径
char pwd[basesize];
char pwdenv[basesize];

//最后一次运行的返回值
int lastcode=0;

// 全局变量与重定向有关
#define NoneRedir    0
#define InputRedir   1
#define OutputRedir  2
#define AppendRedir  3

int Redir = NoneRedir;
char* FileName = nullptr;
#define Trim(pos) do{ \
    while(isspace(*pos)) pos++;\
}while(0);


enum RedirError
{
    InReDirError =1,
    OutReDirError =2,
    AppReDirError =3,
    FileNameError=4
};





void InitEnv()
{
    extern char** environ;
    int index=0;
    while(environ[index])
    {
        Genv[index]=new char[strlen(environ[index])+1];
        strncpy(Genv[index],environ[index],strlen(environ[index])+1);
        index++;
    }
    Genv[index]=nullptr;

}

std::string GetPwd()
{
    if(nullptr==getcwd(pwd,sizeof(pwd))) return "None";
    snprintf(pwdenv,sizeof(pwdenv),"PWD=%s",pwd);
    putenv(pwdenv);
    return pwd;
}


std::string GetLastDir()
{
    std::string cur=GetPwd();
    if(cur=="" || cur=="None") return cur;
    size_t pos=cur.rfind("/");
    return cur.substr(pos+1);
}


std::string GetHostName()
{
    std::string Hostname=getenv("HOSTNAME");
    return Hostname.empty()?"None":Hostname;
}


std::string GetUserName()
{
    std::string name=getenv("USER");
    return name.empty()? "None": name;
}



std::string MakeCommandLine()
{
    // [yang@huawei-cloud myshell]$
    char command_line[basesize];
    snprintf(command_line,basesize,"[%s@%s %s]&",GetUserName().c_str(),GetHostName().c_str(),GetLastDir().c_str());
    return command_line;
}


void PrintCommandLine()
{
    printf("%s",MakeCommandLine().c_str());
    fflush(stdout);
}

bool GetCommandLine(char command_buffer[],int size)
{
    // 我们认为：我们要将用户输入的命令行，当做一个完整的字符串
    // "ls -a -l -n"
    
    char* result=fgets(command_buffer,size,stdin);
    
//    printf("%p\n %p\n",result,command_buffer);

    if(!result)
    {
        return false;
    }
    command_buffer[strlen(command_buffer)-1]=0; // 将结尾的回车设为0
    if(strlen(command_buffer)==0) return false;
    return true;
}

void ParseCommand(char command_buffer[])
{
   memset(Gargv,0,sizeof(Gargv));  // 清空之前的代码痕迹
   Gargc=0;
   
   //"ls -l -a -n"
   
   const char* const sep=" ";
    Gargv[Gargc++]=strtok(command_buffer,sep);  // 将第一个空格前的值切给Gargv[0];

    // 将剩下的值依次切给Gargv, 这里=的含义是 将切完的Gargv的值转为bool 进行比较
    while((bool)(Gargv[Gargc++]=strtok(nullptr,sep)));
    Gargc--;
}

void ResetCommandLine()
{
    memset(Gargv,0,sizeof(Gargv));
    Gargc=0;
    Redir=0;
    FileName=nullptr; 
}

void ParseRedir(char command_buffer[],int len)
{
    int end=len;
    while(end>=0)
    {
        if(command_buffer[end]=='<')
        {
            Redir=InputRedir;
            command_buffer[end]=0;
            FileName=&command_buffer[end]+1;
            Trim(FileName);  // 文件名去空格
            break;
        }
        else if(command_buffer[end]=='>')
        {
            if(command_buffer[end-1]=='>')
            {
                Redir=AppendRedir;
                command_buffer[end-1]=0;
                command_buffer[end]=0;
                FileName=&command_buffer[end]+1;
                Trim(FileName);
                break;
            }
            else 
            {
                Redir=OutputRedir;
                command_buffer[end]=0;
                FileName=&command_buffer[end]+1;
                Trim(FileName);  
                break;
            }
        }
        else end--;
    }

}
    

void ParseCommandLine(char command_buffer[],int len)
{
   ResetCommandLine();
   ParseRedir(command_buffer,len);
   ParseCommand(command_buffer);
}

void DoRedir()
{
    if(Redir==NoneRedir) return;
    if(FileName==nullptr) exit(FileNameError); 
    
    if(Redir==InputRedir)
    {
            int fd=open(FileName,O_RDONLY);
             if(fd<0)
             {
                 exit(InReDirError);
             }
             else dup2(fd,0);
    }
    else if(Redir==OutputRedir)
    {
        int fd=open(FileName,O_CREAT | O_WRONLY | O_TRUNC,0666);
        if(fd<0) exit(OutReDirError);
        else dup2(fd,1);
    }
    else if(Redir==AppendRedir)
    {
        int fd=open(FileName,O_CREAT | O_WRONLY  | O_APPEND,0666);
        if(fd<0) exit(AppReDirError);
        else dup2(fd,1);
    }
}
bool ExecuteCommand()
{
    // 让子进程执行
    pid_t id=fork();
    if(id<0)  return false;
    if(id==0) 
    {
        // 子进程部分
        // 重定向
         DoRedir();
        // 1.执行命令
         execvpe(Gargv[0],Gargv,Genv);
       // execve(Gargv[0],Gargv,Genv);
        //2.退出  正常是不会返回的，如果执行了下面的就说明出错了，所以退出值为1
        exit(1);
    }
    int status=0;
    pid_t rid=waitpid(id,&status,0);
    if(rid>0)
    {
        if(WIFEXITED(status))
        {
            lastcode=WEXITSTATUS(status);
        }
        else 
        {
            lastcode=100;
        }
         return true;
    }
     return false;
}

void AddEnv(const char* item)
{
    int index=0;
    while(Genv[index])
    {
       index++; 
    }
    Genv[index]=new char[strlen(item)+1];
    strncpy(Genv[index],item,strlen(item)+1);
    Genv[++index]=nullptr;
}


bool CheckAndExecBuilCommand()
{
    if(strcmp(Gargv[0],"cd")==0)
    {
        if(Gargc==2)
        {
           chdir(Gargv[1]);
           lastcode=0;
        }
          else 
         {
             lastcode=1;
         }
         return true;
    }
    else if(strcmp(Gargv[0],"export")==0)
    {
        // export 也是内建命令 因为子进程改变了，父进程不会改变
        if(Gargc==2)
        {
            AddEnv(Gargv[1]);
            lastcode=0;
        }
        else 
        {
            lastcode=1;
        }
        return true;
    }
    else if(strcmp(Gargv[0],"env")==0)
    {
        for(int i=0;Genv[i];i++)
        {
            printf("%s\n",Genv[i]);
        }
        lastcode=0;
        return true;
    }
    else if(strcmp(Gargv[0],"echo")==0)
    {
        if(Gargc==2)
        {
            // echo $
            // echo $PATH
            // echo hello
            if(Gargv[1][0]=='$')
            {
                if(Gargv[1][1]=='?')
                {
                    printf("%d\n",lastcode);
                    lastcode=0;
                }
                else 
                {
                    char* temp=&Gargv[1][1];
                    printf("%s\n",getenv(temp));
                    lastcode=0;
                }
            }
            else 
            {
                printf("%s\n",Gargv[1]);
                lastcode=0;
            }
        }
        else 
        {
            lastcode=3;
        }
        return true;
    }
    return false;
}


int main()
{
    InitEnv();
    char command_buffer[basesize];

      while(true)
    {
        PrintCommandLine();  // 1. 命令行提示符
        // command_buffer -> output
        
        if(!GetCommandLine(command_buffer,basesize))   //2.获取用户命令
        {
            continue;
        }
        
        ParseCommandLine(command_buffer,strlen(command_buffer));  // 3.分析命令
        if(CheckAndExecBuilCommand())  // 内建指令，因为shell有部分指令只能又自己调用，而这类指令我们称之为内建指,这里就是检查输入是否为内建指令
        {
            continue;
        }
        ExecuteCommand() ;   //4. 执行命令
    }
    return 0;
}
